import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

/**
 * 350. 两个数组的交集 II
 * https://leetcode-cn.com/problems/intersection-of-two-arrays-ii/
 */
public class Solutions_350 {
    public static void main(String[] args) {
        int[] nums1 = {1, 2, 2, 1}, nums2 = {2, 2};  // output: {2, 2}
//        int[] nums1 = {4, 9, 5}, nums2 = {9, 4, 9, 8, 4};  // output: {4, 9}

        int[] result = intersect2(nums1, nums2);
        System.out.println(Arrays.toString(result));
    }

    /**
     * 解法一：排序
     */
    public static int[] intersect(int[] nums1, int[] nums2) {
        // 对两个数组都进行排序
        Arrays.sort(nums1);
        Arrays.sort(nums2);
        // 交集的个数，取决于最短长度的数组
        int[] res = new int[Math.min(nums1.length, nums2.length)];
        int i = 0, j = 0, k = 0;
        // 同时遍历两个数组
        while (i < nums1.length && j < nums2.length) {
            if (nums1[i] == nums2[j]) {
                // 子集
                res[k] = nums1[i];
                i++;
                j++;
                k++;
            } else if (nums1[i] < nums2[j]) {
                i++;
            } else {
                j++;
            }
        }
        // 共有 k 个交集
        return Arrays.copyOf(res, k);
    }

    /**
     * 解法二：哈希表
     */
    public static int[] intersect2(int[] nums1, int[] nums2) {
        if (nums2.length > nums1.length) {
            return intersect2(nums2, nums1);
        }
        Map<Integer, Integer> map = new HashMap<>();
        // map 中存储长度较大的数组里的元素
        for (int num : nums1) {
            map.put(num, map.getOrDefault(num, 0) + 1);
        }
        // 交集的个数，取决于最短长度的数组
        int[] res = new int[Math.min(nums1.length, nums2.length)];
        int k = 0;
        for (int num : nums2) {
            int count = map.getOrDefault(num, -1);
            if (count > 0) {
                res[k] = num;
                k++;
                // 数量减 1
                map.put(num, count - 1);
            }
        }
        // 共有 k 个交集
        return Arrays.copyOf(res, k);
    }
}
